import torch
import jax
print(jax.default_backend())
print(torch.cuda.is_available())